Python drawing program